#!/bin/bash
# Users can modify the following paths
DATASET_DIR="/backup/data/Audioset/zelin"
WORKSPACE="/backup/data/Audioset/zelin/workspaces/audioset_tagging"

# ============ Train & Inference ============

MODEL_TYPE="Cnn14Bert"
CUDA_VISIBLE_DEVICES=2 python3 pytorch/main_bimodal.py train --workspace=$WORKSPACE --data_type='balanced_train' --window_size=1024 --hop_size=160 --mel_bins=64 --fmin=50 --fmax=8000 --model_type=$MODEL_TYPE --bert_model_type="prajjwal1/bert-medium" --loss_type='clip_bce' --balanced='balanced' --batch_size=32 --max_text_nums 128 --learning_rate=2e-4 --resume_checkpoint_path='/home/zhiling/py3_workspace/audioset_tagging_cnn/Cnn14_mAP=0.431.pth' --save_interval 10000 --aug_prob 0.5 --early_stop=1000000 --cuda

# Plot statistics
python3 -u utils/plot_statistics_bimodal.py plot --dataset_dir=$DATASET_DIR --workspace=$WORKSPACE --select=1_bal_bert --batch_size 32 --window_size 1024 --bert_model_type prajjwal1/bert-medium --max_text_nums 128